import { Graph } from "@antv/x6";
import "@antv/x6-vue-shape";
// import { ModelRect } from "./nodeTypes";
import VueNode from "./vue-node.vue";
import VueEditableNode from "./vue-editable-node.vue";
import VueFlowNode from "./vue-flow-node.vue";

export default function registerNodes() {
  // Graph.registerNode("model-rect", ModelRect);
  Graph.registerVueComponent("vue-node", {
    template: `<vue-node></vue-node>`,
    components: {
      VueNode,
    },
  });
  Graph.registerVueComponent("vue-editable-node", {
    template: `<vue-editable-node></vue-editable-node>`,
    components: {
      VueEditableNode,
    },
  });
  Graph.registerVueComponent("vue-flow-node", {
    template: `<vue-flow-node></vue-flow-node>`,
    components: {
      VueFlowNode,
    },
  });
}
